/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file           : usbd_storage_if.c
  * @version        : v2.0_Cube
  * @brief          : Memory management layer.
  ******************************************************************************
  * @attention
  *
  * <h2><center>&copy; Copyright (c) 2024 STMicroelectronics.
  * All rights reserved.</center></h2>
  *
  * This software component is licensed by ST under Ultimate Liberty license
  * SLA0044, the "License"; You may not use this file except in compliance with
  * the License. You may obtain a copy of the License at:
  *                             www.st.com/SLA0044
  *
  ******************************************************************************
  */
/* USER CODE END Header */

/* Includes ------------------------------------------------------------------*/
#include "usbd_storage_if.h"

/* USER CODE BEGIN INCLUDE */
#include "bsp_driver_sd.h"
#include "sdio.h"
#if UDISK_INCLUDE_W25QXX
#include "W25QXX.h"
#endif
/* USER CODE END INCLUDE */

/* Private typedef -----------------------------------------------------------*/
/* Private define ------------------------------------------------------------*/
/* Private macro -------------------------------------------------------------*/


/* USER CODE BEGIN PV */
/* Private variables ---------------------------------------------------------*/

/* USER CODE END PV */

/** @addtogroup STM32_USB_OTG_DEVICE_LIBRARY
  * @brief Usb device.
  * @{
  */

/** @defgroup USBD_STORAGE
  * @brief Usb mass storage device module
  * @{
  */

/** @defgroup USBD_STORAGE_Private_TypesDefinitions
  * @brief Private types.
  * @{
  */

/* USER CODE BEGIN PRIVATE_TYPES */

/* USER CODE END PRIVATE_TYPES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Defines
  * @brief Private defines.
  * @{
  */
#if UDISK_INCLUDE_W25QXX
#define W25Q64_LUN_NUM                   0
#define SDCard_LUN_NUM                   1
#define STORAGE_LUN_NBR                  2
#else
#define SDCard_LUN_NUM                   0
#define STORAGE_LUN_NBR                  1
#endif
/* USER CODE BEGIN PRIVATE_DEFINES */

/* USER CODE END PRIVATE_DEFINES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Macros
  * @brief Private macros.
  * @{
  */

/* USER CODE BEGIN PRIVATE_MACRO */

/* USER CODE END PRIVATE_MACRO */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Variables
  * @brief Private variables.
  * @{
  */

/* USER CODE BEGIN INQUIRY_DATA_FS */
/** USB Mass storage Standard Inquiry Data. */
const int8_t STORAGE_Inquirydata_FS[] = {/* 72 */
  /* LUN 0 */
  0x00,  // Peripheral Device Type (PDT): 0x00 表示直接访问设备（如硬盘）
  0x80,  // RMB (Removable Medium Bit): 0x80 表示可移动介质
  0x02,  // Version: 0x02 表示SCSI-2兼容
  0x02,  // Response Data Format: 0x02 表示标准响应格式
  (STANDARD_INQUIRY_DATA_LEN - 5),  // Additional Length: 响应数据的长度减去5
  0x00, 0x00, 0x00,  // Reserved
  'S', 'T', 'M', ' ', ' ', ' ', ' ', ' ',  // Manufacturer: 制造商名称（8字节）
  'P', 'r', 'o', 'd', 'u', 'c', 't', ' ',  // Product: 产品名称（16字节）
  ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
  '0', '.', '0' ,'1',  // Version: 版本号（4字节）
#if UDISK_INCLUDE_W25QXX
  /* LUN 1 */
  0x00,  // Peripheral Device Type (PDT): 0x00 表示直接访问设备（如硬盘）
  0x80,  // RMB (Removable Medium Bit): 0x80 表示可移动介质
  0x02,  // Version: 0x02 表示SCSI-2兼容
  0x02,  // Response Data Format: 0x02 表示标准响应格式
  (STANDARD_INQUIRY_DATA_LEN - 5),  // Additional Length: 响应数据的长度减去5
  0x00, 0x00, 0x00,  // Reserved
  'S', 'T', 'M', ' ', ' ', ' ', ' ', ' ',  // Manufacturer: 制造商名称（8字节）
  'S', 'D', 'C', 'a', 'r', 'd', ' ', ' ',  // Product: 产品名称（16字节）
  ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
  '0', '.', '0' ,'1',  // Version: 版本号（4字节）
#endif
};
/* USER CODE END INQUIRY_DATA_FS */

/* USER CODE BEGIN PRIVATE_VARIABLES */

/* USER CODE END PRIVATE_VARIABLES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Exported_Variables
  * @brief Public variables.
  * @{
  */

extern USBD_HandleTypeDef hUsbDeviceFS;

/* USER CODE BEGIN EXPORTED_VARIABLES */

/* USER CODE END EXPORTED_VARIABLES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_FunctionPrototypes
  * @brief Private functions declaration.
  * @{
  */

static int8_t STORAGE_Init_FS(uint8_t lun);
static int8_t STORAGE_GetCapacity_FS(uint8_t lun, uint32_t *block_num, uint16_t *block_size);
static int8_t STORAGE_IsReady_FS(uint8_t lun);
static int8_t STORAGE_IsWriteProtected_FS(uint8_t lun);
static int8_t STORAGE_Read_FS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len);
static int8_t STORAGE_Write_FS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len);
static int8_t STORAGE_GetMaxLun_FS(void);

/* USER CODE BEGIN PRIVATE_FUNCTIONS_DECLARATION */

/* USER CODE END PRIVATE_FUNCTIONS_DECLARATION */

/**
  * @}
  */

USBD_StorageTypeDef USBD_Storage_Interface_fops_FS =
{
  STORAGE_Init_FS,
  STORAGE_GetCapacity_FS,
  STORAGE_IsReady_FS,
  STORAGE_IsWriteProtected_FS,
  STORAGE_Read_FS,
  STORAGE_Write_FS,
  STORAGE_GetMaxLun_FS,
  (int8_t *)STORAGE_Inquirydata_FS
};

/* Private functions ---------------------------------------------------------*/
/**
  * @brief  Initializes over USB FS IP
  * @param  lun:
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Init_FS(uint8_t lun)
{
  /* USER CODE BEGIN 2 */
  USBD_StatusTypeDef ret = USBD_FAIL;
#if UDISK_INCLUDE_W25QXX
  if(lun == W25Q64_LUN_NUM) //W25QXX
  {
      if(W25QXX_ReadID() == W25Q64ID)
        ret = USBD_OK;
  }
#endif
  if(lun == SDCard_LUN_NUM)
      ret = USBD_OK;
  return ret;
  /* USER CODE END 2 */
}

/**
  * @brief  .
  * @param  lun: .
  * @param  block_num: .
  * @param  block_size: .
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_GetCapacity_FS(uint8_t lun, uint32_t *block_num, uint16_t *block_size)
{
  /* USER CODE BEGIN 3 */
  USBD_StatusTypeDef ret = USBD_FAIL;
  switch (lun)
  {
#if UDISK_INCLUDE_W25QXX
      case W25Q64_LUN_NUM:
        *block_num  = W25Q64_BLK_NBR;
        *block_size = W25Q64_BLK_SIZ;
        ret = (USBD_OK);
        break;
#endif
      case SDCard_LUN_NUM:
      {
        HAL_SD_CardInfoTypeDef info;
        if(HAL_SD_GetCardState(&hsd) ==  HAL_SD_CARD_TRANSFER)
        {
          HAL_SD_GetCardInfo(&hsd, &info);
          *block_num = info.LogBlockNbr;
          *block_size = info.LogBlockSize;
          ret = USBD_OK;
        }
      }
      break;
      default: break;
  }
  return ret;
  /* USER CODE END 3 */
}

/**
  * @brief  .
  * @param  lun: .
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_IsReady_FS(uint8_t lun)
{
  /* USER CODE BEGIN 4 */
  return (USBD_OK);
  /* USER CODE END 4 */
}

/**
  * @brief  .
  * @param  lun: .
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_IsWriteProtected_FS(uint8_t lun)
{
  /* USER CODE BEGIN 5 */
  return (USBD_OK);
  /* USER CODE END 5 */
}

/**
  * @brief  .
  * @param  lun: .
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Read_FS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len)
{
  /* USER CODE BEGIN 6 */
  USBD_StatusTypeDef ret = USBD_FAIL;
  switch (lun)
  {
#if UDISK_INCLUDE_W25QXX
    case W25Q64_LUN_NUM:
    {
      uint32_t start_address = blk_addr * W25Q64_BLK_SIZ; //起始块地址
      for (uint16_t i = 0; i < blk_len; i++)
      {
          W25QXX_Read(buf, start_address + i*W25Q64_BLK_SIZ , W25Q64_BLK_SIZ);
      }
      ret = USBD_OK;
    }
    break;
#endif
    case SDCard_LUN_NUM:
      if( HAL_SD_ReadBlocks(&hsd, buf, blk_addr,  blk_len, HAL_MAX_DELAY) == HAL_OK )
      {
        ret = USBD_OK;
        while(HAL_SD_GetState(&hsd) == HAL_SD_STATE_BUSY);
        while( HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER );
      }
    break;
    default: break;
  }
  return ret;
  /* USER CODE END 6 */
}

/**
  * @brief  .
  * @param  lun: .
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Write_FS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len)
{
  /* USER CODE BEGIN 7 */
  USBD_StatusTypeDef ret = USBD_FAIL;
  switch (lun)
  {
#if UDISK_INCLUDE_W25QXX
    case W25Q64_LUN_NUM:
      {
        uint32_t start_address = blk_addr * W25Q64_BLK_SIZ; //算出写入的起始地址
        W25QXX_Write(buf, start_address, blk_len*W25Q64_BLK_SIZ);
        ret = USBD_OK;
      }
      break;
#endif
    case SDCard_LUN_NUM:
      if( HAL_SD_WriteBlocks(&hsd, buf, blk_addr, blk_len, HAL_MAX_DELAY) == HAL_OK )
      {
          ret = USBD_OK;
          while(HAL_SD_GetState(&hsd) == HAL_SD_STATE_BUSY);
          while( HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER );
      }
      break;
    default:break;
  }
  return ret;
  /* USER CODE END 7 */
}

/**
  * @brief  .
  * @param  None
  * @retval .
  */
int8_t STORAGE_GetMaxLun_FS(void)
{
  /* USER CODE BEGIN 8 */
  return (STORAGE_LUN_NBR - 1);
  /* USER CODE END 8 */
}

/* USER CODE BEGIN PRIVATE_FUNCTIONS_IMPLEMENTATION */

/* USER CODE END PRIVATE_FUNCTIONS_IMPLEMENTATION */

/**
  * @}
  */

/**
  * @}
  */

/************************ (C) COPYRIGHT STMicroelectronics *****END OF FILE****/
